import os

import fasttext
import jieba

from __00__config import Config


def predict_fun(data):
    config = Config()
    # 获取数据
    text = data['text']
    char_text = ' '.join(list(text))
    word_text = ' '.join(jieba.lcut(text))
    # 加载本地模型
    # 字符级自动微分模型路径
    char_model_path = os.path.join(config.ft_model_save_path, 'model_char_auto.bin')
    # 词语级自动微分模型路径
    word_model_path = os.path.join(config.ft_model_save_path, 'model_word_auto.bin')
    print(data)


if __name__ == '__main__':
    data = {'text': '中药调理月经不调有效果吗？'}
    predict_fun(data)
